NVIDIA_MODELS = {
    "A40-48Q": {"mb": 49152, "max": 1},
    "A40-24Q": {"mb": 24576, "max": 2},
    "A40-16Q": {"mb": 16384, "max": 3},
    "A40-12Q": {"mb": 12288, "max": 4},
    "A40-8Q": {"mb": 8192, "max": 6},
    "A40-6Q": {"mb": 6144, "max": 8},
    "A40-4Q": {"mb": 4096, "max": 12},
    "A40-3Q": {"mb": 3072, "max": 16},
    "A40-2Q": {"mb": 2048, "max": 24},
    "A40-1Q": {"mb": 1024, "max": 32},
    "A10-24Q": {"mb": 24576, "max": 1},
    "A10-12Q": {"mb": 12288, "max": 2},
    "A10-8Q": {"mb": 8192, "max": 3},
    "A10-6Q": {"mb": 6144, "max": 4},
    "A10-4Q": {"mb": 4096, "max": 6},
    "A10-3Q": {"mb": 3072, "max": 8},
    "A10-2Q": {"mb": 2048, "max": 12},
    "A10-1Q": {"mb": 1024, "max": 24},
    "A16-16Q": {"mb": 16384, "max": 1},
    "A16-8Q": {"mb": 8192, "max": 2},
    "A16-4Q": {"mb": 4096, "max": 4},
    "A16-2Q": {"mb": 2048, "max": 8},
    "A16-1Q": {"mb": 1024, "max": 16},
    "RTXA6000-48Q": {"mb": 49152, "max": 1},
    "RTXA6000-24Q": {"mb": 24576, "max": 2},
    "RTXA6000-16Q": {"mb": 16384, "max": 3},
    "RTXA6000-12Q": {"mb": 12288, "max": 4},
    "RTXA6000-8Q": {"mb": 8192, "max": 6},
    "RTXA6000-6Q": {"mb": 6144, "max": 8},
    "RTXA6000-4Q": {"mb": 4096, "max": 12},
    "RTXA6000-3Q": {"mb": 3072, "max": 16},
    "RTXA6000-2Q": {"mb": 2048, "max": 24},
    "RTXA6000-1Q": {"mb": 1024, "max": 32},
    "M60-8Q": {"mb": 8192, "max": 1},
    "M60-4Q": {"mb": 4096, "max": 2},
    "M60-2Q": {"mb": 2048, "max": 4},
    "M60-1Q": {"mb": 1024, "max": 8},
    "M60-0Q": {"mb": 512, "max": 16},
    "M10-8Q": {"mb": 8192, "max": 1},
    "M10-4Q": {"mb": 4096, "max": 2},
    "M10-2Q": {"mb": 2048, "max": 4},
    "M10-1Q": {"mb": 1024, "max": 8},
    "M10-0Q": {"mb": 512, "max": 16},
    "M6-8Q": {"mb": 8192, "max": 1},
    "M6-4Q": {"mb": 4096, "max": 2},
    "M6-2Q": {"mb": 2048, "max": 4},
    "M6-1Q": {"mb": 1024, "max": 8},
    "M6-0Q": {"mb": 512, "max": 16},
    "P100C-12Q": {"mb": 12288, "max": 1},
    "P100C-6Q": {"mb": 6144, "max": 2},
    "P100C-4Q": {"mb": 4096, "max": 3},
    "P100C-2Q": {"mb": 2048, "max": 6},
    "P100C-1Q": {"mb": 1024, "max": 12},
    "P100-16Q": {"mb": 16384, "max": 1},
    "P100-8Q": {"mb": 8192, "max": 2},
    "P100-4Q": {"mb": 4096, "max": 4},
    "P100-2Q": {"mb": 2048, "max": 8},
    "P100-1Q": {"mb": 1024, "max": 16},
    "P100X-16Q": {"mb": 16384, "max": 1},
    "P100X-8Q": {"mb": 8192, "max": 2},
    "P100X-4Q": {"mb": 4096, "max": 4},
    "P100X-2Q": {"mb": 2048, "max": 8},
    "P100X-1Q": {"mb": 1024, "max": 16},
    "P40-24Q": {"mb": 24576, "max": 1},
    "P40-12Q": {"mb": 12288, "max": 2},
    "P40-8Q": {"mb": 8192, "max": 3},
    "P40-6Q": {"mb": 6144, "max": 4},
    "P40-4Q": {"mb": 4096, "max": 6},
    "P40-3Q": {"mb": 3072, "max": 8},
    "P40-2Q": {"mb": 2048, "max": 12},
    "P40-1Q": {"mb": 1024, "max": 24},
    "P6-16Q": {"mb": 16384, "max": 1},
    "P6-8Q": {"mb": 8192, "max": 2},
    "P6-4Q": {"mb": 4096, "max": 4},
    "P6-2Q": {"mb": 2048, "max": 8},
    "P6-1Q": {"mb": 1024, "max": 16},
    "P4-8Q": {"mb": 8192, "max": 1},
    "P4-4Q": {"mb": 4096, "max": 2},
    "P4-2Q": {"mb": 2048, "max": 4},
    "P4-1Q": {"mb": 1024, "max": 8},
    "T4-16Q": {"mb": 16384, "max": 1},
    "T4-8Q": {"mb": 8192, "max": 2},
    "T4-4Q": {"mb": 4096, "max": 4},
    "T4-2Q": {"mb": 2048, "max": 8},
    "T4-1Q": {"mb": 1024, "max": 16},
    "V100X-16Q": {"mb": 16384, "max": 1},
    "V100X-8Q": {"mb": 8192, "max": 2},
    "V100X-4Q": {"mb": 4096, "max": 4},
    "V100X-2Q": {"mb": 2048, "max": 8},
    "V100X-1Q": {"mb": 1024, "max": 16},
    "V100DX-32Q": {"mb": 32768, "max": 1},
    "V100DX-16Q": {"mb": 16384, "max": 2},
    "V100DX-8Q": {"mb": 8192, "max": 4},
    "V100DX-4Q": {"mb": 4096, "max": 8},
    "V100DX-2Q": {"mb": 2048, "max": 16},
    "V100DX-1Q": {"mb": 1024, "max": 32},
    "V100-16Q": {"mb": 16384, "max": 1},
    "V100-8Q": {"mb": 8192, "max": 2},
    "V100-4Q": {"mb": 4096, "max": 4},
    "V100-2Q": {"mb": 2048, "max": 8},
    "V100-1Q": {"mb": 1024, "max": 16},
    "V100D-32Q": {"mb": 32768, "max": 1},
    "V100D-16Q": {"mb": 16384, "max": 2},
    "V100D-8Q": {"mb": 8192, "max": 4},
    "V100D-4Q": {"mb": 4096, "max": 8},
    "V100D-2Q": {"mb": 2048, "max": 16},
    "V100D-1Q": {"mb": 1024, "max": 32},
    "V100S-32Q": {"mb": 32768, "max": 1},
    "V100S-16Q": {"mb": 16384, "max": 2},
    "V100S-8Q": {"mb": 8192, "max": 4},
    "V100S-4Q": {"mb": 4096, "max": 8},
    "V100S-2Q": {"mb": 2048, "max": 16},
    "V100S-1Q": {"mb": 1024, "max": 32},
    "V100L-16Q": {"mb": 16384, "max": 1},
    "V100L-8Q": {"mb": 8192, "max": 2},
    "V100L-4Q": {"mb": 4096, "max": 4},
    "V100L-2Q": {"mb": 2048, "max": 8},
    "V100L-1Q": {"mb": 1024, "max": 16},
    "RTX8000-48Q": {"mb": 49152, "max": 1},
    "RTX8000-24Q": {"mb": 24576, "max": 2},
    "RTX8000-16Q": {"mb": 16384, "max": 3},
    "RTX8000-12Q": {"mb": 12288, "max": 4},
    "RTX8000-8Q": {"mb": 8192, "max": 6},
    "RTX8000-6Q": {"mb": 6144, "max": 8},
    "RTX8000-4Q": {"mb": 4096, "max": 12},
    "RTX8000-3Q": {"mb": 3072, "max": 16},
    "RTX8000-2Q": {"mb": 2048, "max": 24},
    "RTX8000-1Q": {"mb": 1024, "max": 32},
    "RTX8000P-48Q": {"mb": 49152, "max": 1},
    "RTX8000P-24Q": {"mb": 24576, "max": 2},
    "RTX8000P-16Q": {"mb": 16384, "max": 3},
    "RTX8000P-12Q": {"mb": 12288, "max": 4},
    "RTX8000P-8Q": {"mb": 8192, "max": 6},
    "RTX8000P-6Q": {"mb": 6144, "max": 8},
    "RTX8000P-4Q": {"mb": 4096, "max": 12},
    "RTX8000P-3Q": {"mb": 3072, "max": 16},
    "RTX8000P-2Q": {"mb": 2048, "max": 24},
    "RTX8000P-1Q": {"mb": 1024, "max": 32},
    "RTX6000-24Q": {"mb": 24576, "max": 1},
    "RTX6000-12Q": {"mb": 12288, "max": 2},
    "RTX6000-8Q": {"mb": 8192, "max": 3},
    "RTX6000-6Q": {"mb": 6144, "max": 4},
    "RTX6000-4Q": {"mb": 4096, "max": 6},
    "RTX6000-3Q": {"mb": 3072, "max": 8},
    "RTX6000-2Q": {"mb": 2048, "max": 12},
    "RTX6000-1Q": {"mb": 1024, "max": 24},
    "RTX6000P-24Q": {"mb": 24576, "max": 1},
    "RTX6000P-12Q": {"mb": 12288, "max": 2},
    "RTX6000P-8Q": {"mb": 8192, "max": 3},
    "RTX6000P-6Q": {"mb": 6144, "max": 4},
    "RTX6000P-4Q": {"mb": 4096, "max": 6},
    "RTX6000P-3Q": {"mb": 3072, "max": 8},
    "RTX6000P-2Q": {"mb": 2048, "max": 12},
    "RTX6000P-1Q": {"mb": 1024, "max": 24},
}
